package com.nbsaas.boot.product.rest.convert;

import com.nbsaas.boot.product.data.entity.ProductFile;
import com.nbsaas.boot.product.api.domain.simple.ProductFileSimple;

import com.nbsaas.boot.rest.api.Converter;
import java.util.HashMap;
import java.util.Map;

/**
* 商品文件-列表对象转换器
*/

public class ProductFileSimpleConvert implements Converter<ProductFileSimple, ProductFile> {




@Override
public ProductFileSimple convert(ProductFile source) {
    ProductFileSimple result = new ProductFileSimple();


               try {
                   if(source.getProduct()!=null){
                       result.setProduct(source.getProduct().getId());
                   }
                } catch (Exception ignored) { }

                result.setId(source.getId());

                try {
                    if(source.getProduct()!=null){
                        result.setProductName(source.getProduct().getName());
                    }
                } catch (Exception ignored) { }

                if(source.getFileType()!=null){
                   Map<Integer,String> FileTypeMap=new HashMap<>();
                       FileTypeMap.put(1,"图片");
                       FileTypeMap.put(2,"视频");
                       FileTypeMap.put(3,"文件");
                   String label=  FileTypeMap.get(source.getFileType());
                   result.setFileTypeName(label);
                }
                result.setFileType(source.getFileType());
                result.setUrl(source.getUrl());


    return result;
}

}